package com.learn.swagger2_gradle.config;

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.env.Environment;
import org.springframework.core.env.Profiles;
import springfox.documentation.builders.PathSelectors;
import springfox.documentation.builders.RequestHandlerSelectors;
import springfox.documentation.service.ApiInfo;
import springfox.documentation.service.Contact;
import springfox.documentation.spi.DocumentationType;
import springfox.documentation.spring.web.plugins.Docket;
import springfox.documentation.swagger2.annotations.EnableSwagger2;

import java.util.ArrayList;

/**
 * @author: rensiyu
 * @create: 2021-07-19 15:00
 * @description:
 * @Version 1.0
 **/

@Configuration
@EnableSwagger2   // 开启Swagger2
public class SwaggerConfig {

    // 配置了swagger的Docket 的 bean 实例
    @Bean
    public Docket docket() {
        return new Docket(DocumentationType.SWAGGER_2)
                .apiInfo(apiInfo())

                // .enable(false) 是否要启动swagger

                .select()

                // RequestHandlerSelectors, 配置要扫描的包
                // basePackage(): 指定要扫描的包
                // any(): 扫描全部
                // none(): 都不扫描
                // withClassAnnotation : 扫描类上的注解, 参数是一个注解的反射对象
                // 例如：withClassAnnotation(RestController.class) 只扫描类上有@RestController的生成文档
                // withMethodAnnotation: 扫描方法上的注解, 参数是一个注解的反射对象
                .apis(RequestHandlerSelectors.basePackage("com.kuang.swagger.controller"))

                // paths(): 过滤什么路径
                .paths(PathSelectors.ant("/kuang/**"))
                .build();  //build
    }

    /**
     * 在方法中加入形参Environment获取application.properties中的设置的环境
     *
     * @param environment
     * @return
     */
    @Bean
    public Docket docket(Environment environment) {
        // 设置要显示的Swagger环境
        Profiles profiles = Profiles.of("dev", "test");
        // 通过 environment.acceptsProfiles(profiles) 判断是否处在自己设定的环境当中
        boolean flag = environment.acceptsProfiles(profiles);


        return new Docket(DocumentationType.SWAGGER_2)
                .apiInfo(apiInfo())
                //此处判断是生产环境 flag 就是为true 就可以访问Swagger
                .enable(flag)

                .select()
                .apis(RequestHandlerSelectors.basePackage("com.kuang.swagger.controller"))
                // .paths(PathSelectors.ant("/kuang/**"))
                .build();  //build
    }

    @Bean
    public Docket docket1() {
        return new Docket(DocumentationType.SWAGGER_2).groupName("分组A");
    }


    // 配置 swagger 信息 = apiInfo
    private ApiInfo apiInfo() {

        // 作者信息
        Contact contact = new Contact("秦疆", "https://blog.kuangstudy.com/", "24736743@qq.com");
        return new ApiInfo("狂神的SwaggerAPI文档",
                "即时在小的帆也能远航",
                "v1.0",
                "http://blog.kuangstudy.com/",
                contact,
                "Apache 2.0",
                "http://www.apache.org/licenses/LICENSE-2.0",
                new ArrayList()
        );
    }


}
